package com.cotasul.tela.datamodel;

import java.io.Serializable;
import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import com.cotasul.bean.ProdutoBean;

public class ProdutoDataModel extends ListDataModel<ProdutoBean> implements
		SelectableDataModel<ProdutoBean>, Serializable {

	private static final long serialVersionUID = -4191930492802741250L;

	public ProdutoDataModel(List<ProdutoBean> data) {
		super(data);
	}

	@SuppressWarnings("unchecked")
	@Override
	public ProdutoBean getRowData(String linha) {
		List<ProdutoBean> prods = (List<ProdutoBean>) getWrappedData();
		for (ProdutoBean prod : prods) {
			if (prod.getCodigo().equals(linha)) {
				return prod;
			}
		}
		return null;
	}

	@Override
	public Object getRowKey(ProdutoBean linha) {
		return linha.getCodigo();
	}

}
